package com.lx.intercepted;

import com.lx.annotation.MyCache;
import com.lx.annotation.limit;
import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletResponse;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;
import org.springframework.web.method.HandlerMethod;
import org.springframework.web.servlet.HandlerInterceptor;
import org.springframework.web.servlet.ModelAndView;


import java.util.concurrent.TimeUnit;

/**
 * 雒鑫
 */
@Component
public class LimitInterceptor implements HandlerInterceptor {
    @Autowired
    private RedisTemplate redisTemplate;

    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        //1.判断当前的请求是否是一个方法
        if (handler instanceof HandlerMethod) {
            HandlerMethod handlerMethod = (HandlerMethod) handler;
            //获取目标方法上是否有指定注解
            limit limit = handlerMethod.getMethodAnnotation(limit.class);
            if (limit == null) {
                //说明方法上没有@limt注解
                return true;
            }
            //执行到此处 说明方法上有limit注解，校验这个请求是否刷接口
            //获取当前请求地址
            String url = String.valueOf(request.getRequestURL());
            //将请求的地址与注解当中的key进行拼接 存储到redis当中充当键的值 再将次数增加1次
            String key = limit.key() + url;
            Long count = redisTemplate.opsForValue().increment(key);
            if (count == 1) {
                redisTemplate.expire(key, limit.timeOut(), TimeUnit.SECONDS);
            }
            if (count > limit.sum()) {
                System.out.println("当前请求已经超过最大限制次数，请稍后再访问");
                response.setStatus(511);
                return false;
            }
        }
        //如果当前的请求不是一个方法 那么直接放行
        return true;
    }


}

